// This file is part of RSS Guard.

//
// Copyright (C) 2011-2017 by Martin Rotter <rotter.martinos@gmail.com>
// Copyright (C) 2010-2014 by David Rosca <nowrep@gmail.com>
//
// RSS Guard is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// RSS Guard is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with RSS Guard. If not, see <http://www.gnu.org/licenses/>.

#include "gui/treewidget.h"

#include <QMouseEvent>

TreeWidget::TreeWidget(QWidget* parent)
  : QTreeWidget(parent), m_refreshAllItemsNeeded(true), m_showMode(ItemsCollapsed) {
  connect(this, SIGNAL(itemChanged(QTreeWidgetItem*,int)), this, SLOT(sheduleRefresh()));
}

void TreeWidget::clear() {
  QTreeWidget::clear();
  m_allTreeItems.clear();
}

void TreeWidget::sheduleRefresh() {
  m_refreshAllItemsNeeded = true;
}

void TreeWidget::addTopLevelItem(QTreeWidgetItem* item) {
  m_allTreeItems.append(item);
  QTreeWidget::addTopLevelItem(item);
}

void TreeWidget::addTopLevelItems(const QList<QTreeWidgetItem*>& items) {
  m_allTreeItems.append(items);
  QTreeWidget::addTopLevelItems(items);
}

void TreeWidget::insertTopLevelItem(int index, QTreeWidgetItem* item) {
  m_allTreeItems.append(item);
  QTreeWidget::insertTopLevelItem(index, item);
}

void TreeWidget::insertTopLevelItems(int index, const QList<QTreeWidgetItem*>& items) {
  m_allTreeItems.append(items);
  QTreeWidget::insertTopLevelItems(index, items);
}

void TreeWidget::mousePressEvent(QMouseEvent* event) {
  if (event->modifiers() == Qt::ControlModifier) {
    emit itemControlClicked(itemAt(event->pos()));
  }

  if (event->buttons() == Qt::MiddleButton) {
    emit itemMiddleButtonClicked(itemAt(event->pos()));
  }

  QTreeWidget::mousePressEvent(event);
}

void TreeWidget::iterateAllItems(QTreeWidgetItem* parent) {
  int count = parent ? parent->childCount() : topLevelItemCount();

  for (int i = 0; i < count; i++) {
    QTreeWidgetItem* item = parent ? parent->child(i) : topLevelItem(i);

    if (item->childCount() == 0) {
      m_allTreeItems.append(item);
    }

    iterateAllItems(item);
  }
}

QList<QTreeWidgetItem*> TreeWidget::allItems() {
  if (m_refreshAllItemsNeeded) {
    m_allTreeItems.clear();
    iterateAllItems(0);
    m_refreshAllItemsNeeded = false;
  }

  return m_allTreeItems;
}

void TreeWidget::filterString(const QString& string) {
  QList<QTreeWidgetItem*> _allItems = allItems();
  QList<QTreeWidgetItem*> parents;
  bool stringIsEmpty = string.isEmpty();

  foreach (QTreeWidgetItem* item, _allItems) {
    bool containsString = stringIsEmpty || item->text(0).contains(string, Qt::CaseInsensitive);

    if (containsString) {
      item->setHidden(false);

      if (item->parent()) {
        if (!parents.contains(item->parent())) {
          parents << item->parent();
        }
      }
    }
    else {
      item->setHidden(true);

      if (item->parent()) {
        item->parent()->setHidden(true);
      }
    }
  }

  for (int i = 0; i < parents.size(); ++i) {
    QTreeWidgetItem* parentItem = parents.at(i);

    parentItem->setHidden(false);

    if (stringIsEmpty) {
      parentItem->setExpanded(m_showMode == ItemsExpanded);
    }
    else {
      parentItem->setExpanded(true);
    }

    if (parentItem->parent() && !parents.contains(parentItem->parent())) {
      parents << parentItem->parent();
    }
  }
}

bool TreeWidget::appendToParentItem(const QString& parentText, QTreeWidgetItem* item) {
  QList<QTreeWidgetItem*> list = findItems(parentText, Qt::MatchExactly);

  if (list.count() == 0) {
    return false;
  }

  QTreeWidgetItem* parentItem = list.at(0);

  if (!parentItem) {
    return false;
  }

  m_allTreeItems.append(item);
  parentItem->addChild(item);
  return true;
}

bool TreeWidget::appendToParentItem(QTreeWidgetItem* parent, QTreeWidgetItem* item) {
  if (!parent || parent->treeWidget() != this) {
    return false;
  }

  m_allTreeItems.append(item);
  parent->addChild(item);
  return true;
}

bool TreeWidget::prependToParentItem(const QString& parentText, QTreeWidgetItem* item) {
  QList<QTreeWidgetItem*> list = findItems(parentText, Qt::MatchExactly);

  if (list.count() == 0) {
    return false;
  }

  QTreeWidgetItem* parentItem = list.at(0);

  if (!parentItem) {
    return false;
  }

  m_allTreeItems.append(item);
  parentItem->insertChild(0, item);
  return true;
}

bool TreeWidget::prependToParentItem(QTreeWidgetItem* parent, QTreeWidgetItem* item) {
  if (!parent || parent->treeWidget() != this) {
    return false;
  }

  m_allTreeItems.append(item);
  parent->insertChild(0, item);
  return true;
}

void TreeWidget::deleteItem(QTreeWidgetItem* item) {
  m_refreshAllItemsNeeded = true;
  delete item;
}

void TreeWidget::deleteItems(const QList<QTreeWidgetItem*>& items) {
  m_refreshAllItemsNeeded = true;
  qDeleteAll(items);
}
